# from utils import AIHandler
import time

def main():
    # 创建AIHandler实例 - 使用默认配置
    handler = AIHandler()
    # print("=== 初始化完成 ===")
    # print(f"当前系统提示语: {handler.initial_prompt}\n")
    # time.sleep(1)

    # # 测试基本对话功能
    # print("=== 测试基本对话 ===")
    # response = handler.chat("你好！")
    # print("用户: 你好！")
    # print(f"AI助手: {response}\n")
    # time.sleep(1)

    # # 测试连续对话（保持上下文）
    # print("=== 测试连续对话 ===")
    # response = handler.chat("我们刚才说了什么？")
    # print("用户: 我们刚才说了什么？")
    # print(f"AI助手: {response}\n")
    # time.sleep(1)

    # 测试重置提示语
    print("=== 测试重置提示语 ===")
    handler.reset_prompt("你现在是一个专业的Python编程助手。")
    print("已重置系统提示语为: '你现在是一个专业的Python编程助手。'")
    response = handler.chat("请解释一下Python中的装饰器是什么？")
    print("用户: 请解释一下Python中的装饰器是什么？")
    print(f"AI助手: {response}\n")
    time.sleep(5)

    # 测试清除历史记录
    print("=== 测试清除历史记录 ===")
    handler.flush()
    print("已清除历史记录")
    response = handler.chat("我们之前讨论过装饰器吗？")
    print("用户: 我们之前讨论过装饰器吗？")
    print(f"AI助手: {response}\n")
    time.sleep(1)


if __name__ == "__main__":
    # main() 
    from bilibili_api import live, sync

    room = live.LiveDanmaku(22544798)

    @room.on('DANMU_MSG')
    async def on_danmaku(event):
        # 收到弹幕
        print(event)
        
    sync(room.connect())